# knowledge/database/__init__.py

from .db_manager import DatabaseManager

# 检查并导入其他类（如果存在）
try:
    from .db_query_processor import DatabaseQueryProcessor
except ImportError:
    # 如果文件不存在，创建一个空类占位
    class DatabaseQueryProcessor:
        def __init__(self, db_manager):
            self.db_manager = db_manager

try:
    from .db_schema_generator import DatabaseSchemaGenerator
except ImportError:
    # 如果文件不存在，创建一个空类占位
    class DatabaseSchemaGenerator:
        def __init__(self, db_manager):
            self.db_manager = db_manager

__all__ = [
    "DatabaseManager",
    "DatabaseQueryProcessor", 
    "DatabaseSchemaGenerator"
]